# 全国企业工商信息
data_home_folder = "/media/jie/新加卷/pku_data/industry_info"

cls_explain = [
    "煤制氢，天然气制氢;",
    "生产化工产品时得到氢气，主要有焦炉煤气、氯碱化工、轻烃利用（丙烷脱氢、乙烷裂解）、合成氨、合成甲醇等工业的副产氢;",
    "电解水制氢;",
    "压缩气态储氢,将氢气压缩于高压容器;",
    "低温液态储氢,低温 (-253℃)条件下对氢气进行液化;",
    "液氨/甲醇储氢;",
    "氢化物/LOHC,吸附储氢利用金属合金、碳质材料、有机液体材料、金属框架物等,实现对氢的吸附储氢和释放的实现;",
    "加氢站,加氢站的主要设备包括储氢装置、压缩设备、加注设备、站控系统等;",
    "氢燃料电池汽车;",
    "绿色航运,绿色航运液态燃料,分别为LNG、绿色甲醇、绿氨、液氢;",
    "可持续航煤,航空实现零碳转型。主流生产技术路线,分别是油脂加氢、气化-费托合成、醇制油和合成燃料技术;",
    "绿氢耦合化工，绿氢生产;",
    "氢能炼化行业,加氢裂化工段和加氢精制工段;",
    "氢冶金是钢铁生产实现低碳冶炼的革命性技术，包括：氢气气基竖炉技术;",
    "氢气发电;",
    "建筑领域，天然气掺氢用作家用燃料；氢燃料为建筑物供电供热;",
]

cls_name = [
    "化石燃料制氢",
    "工业副产氢",
    "电解水制氢",
    "压缩气态储氢",
    "低温液态储氢",
    "液氨/甲醇储氢",
    "氢化物/LOHC",
    "加氢站",
    "氢燃料电池汽车",
    "绿色航运",
    "可持续航煤",
    "绿氢耦合化工",
    "氢能炼化行业",
    "氢冶金",
    "氢气发电",
    "建筑领域",
]

industry_explain_template = "{keyword}: {explain}\n"

kv = []
for i in range(len(cls_explain)):
    kv.append(
        industry_explain_template.format(keyword=cls_name[i], explain=cls_explain[i])
    )

kv = "".join(kv)


prompt_cls = """
你是一位氢能领域企业分类的专家。请仔细阅读企业信息：{industry_info}  
接下来，请从 {cls_name} 中选出一个最符合该企业信息的类别，若企业信息不属于任何类别，请返回['其他']。  
下述是这些类别的定义：
{kv}
返回格式：['类别名']
"""
industry_info_template = "{industry_name}的经营范围是{domain};"


def llm_industry_prompt(industry_name, domain):
    industry_info = industry_info_template.format(
        industry_name=industry_name, domain=domain
    )
    return prompt_cls.format(
        industry_info=industry_info, cls_name="、".join(cls_name), kv=kv
    )

hydrogen_kw = [
    "加氢", "储氢"
]